op 	 {
  name: "ConfigureTPUEmbeddingMemory"
  input_arg {
    name: "common_config"
    type: DT_STRING
  }
  output_arg {
    name: "memory_config"
    type: DT_STRING
  }
  is_stateful: true
}
